package com.burst.reptile.common.utils.ip;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.alibaba.fastjson.JSONObject;
import com.burst.reptile.framework.config.CrawlerConfig;

import cn.hutool.core.util.StrUtil;
import cn.hutool.http.HttpUtil;

/**
 * 获取地址类
 * 
 * @author wujin
 * @date: 2020年5月21日 下午3:23:44
 */
public class AddressUtils {
	private static final Logger log = LoggerFactory.getLogger(AddressUtils.class);

	// IP地址查询
	public static final String IP_URL = "http://whois.pconline.com.cn/ipJson.jsp";

	// 未知地址
	public static final String UNKNOWN = "XX XX";

	public static String getRealAddressByIP(String ip) {
		String address = UNKNOWN;
		// 内网不查询
		if (IpUtils.internalIp(ip)) {
			return "内网IP";
		}
		if (CrawlerConfig.getAddressEnabled()) {
			try {
				String rspStr= HttpUtil.get(IP_URL + "?ip=" + ip + "&json=true");
				if (StrUtil.isEmpty(rspStr)) {
					log.error("获取地理位置异常 {}", ip);
					return UNKNOWN;
				}
				JSONObject obj = JSONObject.parseObject(rspStr);
				String region = obj.getString("pro");
				String city = obj.getString("city");
				return String.format("%s %s", region, city);
			} catch (Exception e) {
				log.error("获取地理位置异常 {}", ip);
			}
		}
		return address;
	}
	
	/**
	 * 更具IP获取地址信息
	 * @param ip
	 * @return
	 */
	public static String getAddressByIP(String ip) {
		String address = null;
		// 内网不查询
		if (IpUtils.internalIp(ip)) {
			return address;
		}
		if (CrawlerConfig.getAddressEnabled()) {
			try {
				String rspStr = HttpUtil.get(IP_URL + "?ip=" + ip + "&json=true");
				JSONObject obj = JSONObject.parseObject(rspStr);
				return obj.getString("city");
			} catch (Exception e) {
				log.error("获取地理位置异常 {}", ip);
			}
		}
		return address;
	}
}
